from langgraph.graph import StateGraph
from typing_extensions import TypedDict

class InternalSatate(TypedDict):
    user_query:str
    search_results : str
    llm_response:str
    debug_info:str

class InputSchema(TypedDict):
    user_query:str

class OutputSchema(TypedDict):
    llm_response:str

graph = StateGraph(InternalSatate,input=InputSchema,output=OutputSchema)